﻿package RecursiveSearcher;

import java.util.ArrayList;
import java.util.List;

/**
 * @ClassName LC216
 * @Description 找出所有相加之和为 n 的 k 个数的组合，且满足下列条件：
 *
 * 只使用数字1到9
 * 每个数字 最多使用一次
 * 返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次，组合可以以任何顺序返回。
 *
 *
 *
 * 示例 1:
 *
 * 输入: k = 3, n = 7
 * 输出: [[1,2,4]]
 * 解释:
 * 1 + 2 + 4 = 7
 *
 * @Author ZJX
 * @Date 2024/10/22 21:27
 * @Version 1.0
 */
public class LC216 {
    List<List<Integer>> ret;  // 存储所有符合条件的组合
    List<Integer> path;       // 当前路径

    public List<List<Integer>> combinationSum3(int k, int n) {
        ret = new ArrayList<>();
        path = new ArrayList<>();
        dfs(n, k, 0, 1);  // 从数字1开始搜索，初始和为0
        return ret;
    }

    private void dfs(int targetSum, int k, int sum, int pos) {
        // 剪枝：如果当前和已经超过目标和，直接返回
        if (sum > targetSum) {
            return;
        }

        // 如果当前组合长度等于 k，检查累加和是否等于目标和
        if (path.size() == k) {
            if (sum == targetSum) {
                ret.add(new ArrayList<>(path));  // 保存当前组合
            }
            return;
        }

        // 在 [pos, 9] 范围内循环选择数字
        for (int i = pos; i <= 9 -(k - path.size()) + 1; i++) {
            // 剪枝：如果 sum + i 超过目标和，提前退出循环
            if (sum + i > targetSum) {
                break;
            }

            path.add(i);  // 选择当前数字
            dfs(targetSum, k, sum + i, i + 1);  // 递归调用
            path.remove(path.size() - 1);  // 回溯，撤销选择
        }
    }

}
